1121C - System Testing - CodeForces Solution

#include "bits/stdc++.h"
using namespace std;
typedef long long int ll;
double l[1500];
double r[1500];
int main() {
    int n, k;
    cin >> n >> k;
    vector <int> a(n), t(k, 0);
    for (int i = 0; i < n; ++i) {
        cin >> a[i];
    }
    int ptr = 0;
    int s = 0;
    while (ptr < n) {
        for (int j = 0; j < k && ptr < n; ++j) {
            if (t[j] != 0) {
                t[j]--;
            }
            if (t[j] == 0) {
                t[j] = a[ptr];
                l[ptr] = s;
                r[ptr] = s + a[ptr];
                ptr++;
            }
        }
        s++;
    }
    vector <int> tm;
    for (int i = 0; i < n; ++i)
        tm.push_back(r[i]);
    sort(tm.begin(), tm.end());
    set <int> ans;
    double zn = n;
    for (int i = 0; i < n; ++i) {
        int test = 1;
        for (int j = l[i]; j < r[i]; ++j) {
            double cnt = 0;
            for (int k = 0; k < tm.size(); ++k) {
                if (tm[k] > j)
                    break;
                cnt++;
            }
            double ch = round(100 * (cnt / zn));
            if (ch == test) {
                ans.insert(i);
            }
            ++test;
        }
    }
    cout << ans.size();
}